import argparse
from dataclasses import dataclass


@dataclass
class Config:
    gui_mode: bool
    model_config: str


def parse_arg() -> Config:
    parser = argparse.ArgumentParser(description="AI Chat")

    parser.add_argument('--gui', action='store_true', help='启用 GUI 模式')
    parser.add_argument('--model_config', default='./config/config.toml', help='设置模型配置文件')

    args = parser.parse_args()

    return Config(
        gui_mode=args.gui,
        model_config=args.model_config
    )